const socket = new WebSocket("ws://127.0.0.1:5555/")
export function initWs() {
  // 创建 WebSocket 连接

  // 当连接建立时
  socket.addEventListener("open", function (event) {
    console.log("WebSocket 连接已建立")
    sendMessage("connection")
  })

  // 当接收到消息时
  socket.addEventListener("message", function (event) {
    const data = event.data
    return data
  })

  // 当发生错误时
  socket.addEventListener("error", function (event) {
    console.error("WebSocket 连接发生错误")
  })

  // 当连接关闭时
  socket.addEventListener("close", function (event) {
    console.log("WebSocket 连接已关闭")
  })
  return socket
}

// 发送消息的函数
export function sendMessage(message) {
  // 确保连接已经建立
  if (socket.readyState === WebSocket.OPEN) {
    // 发送文本消息
    socket.send(message)
  } else {
    console.error("WebSocket 连接尚未建立")
  }
}
